list as user input in python